from myclasses.a4_train_methods import get_trained_inferior, get_trained_brain, get_series_trained_brain
from myclasses.a2_load_dataset import get_loaders

version = '20220917'

# loaders prepare
# train_loader, test_loader = get_loaders()
# train_loader, test_loader = get_loaders(load_version=version)

# train inferior
# inferior = get_trained_inferior(train_loader, test_loader)
# inferior = get_trained_inferior(version=version)

# train superior
# brain = get_trained_brain(train_loader, test_loader, inferior_version=version, turns_rate=2)
# brain = get_trained_brain(version=version)

# series train
# brains = get_series_trained_brain(train_loader, test_loader, inferior_version=version, turns_rate=2)
brains = get_series_trained_brain(version=version)
print('training finished')
